package com.loop.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

/**
 * @author zhangxiaolu
 * @since 2023-09-16 20:33
 */
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

    // 私聊思路：使用一个HashMap管理
//    public static final HashMap<String, Channel> clients = new HashMap<String, Channel>();

    //定义一个Channel 组，管理所有的Channel
    // GlobalEventExecutor.INSTANCE 是全局的事件执行器，是一个单例
    private static final ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

//    SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

    /**
     * HandlerAdded 事件，当有新的Channel 加入时，触发；即一旦连接建立第一个被执行
     * 将当前Channel 加入到ChannelGroup 中
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) throws Exception {
        //将当前的Channel 添加到Channel 组中
        // 将该客户加入聊天的信息推送给在线的客户端
        channelGroup.writeAndFlush("[客户端]" + ctx.channel().remoteAddress() + "加入聊天\n");
        channelGroup.add(ctx.channel());
    }

    /**
     * 断开连接，将xx客户端的离线信息推送给在线的客户端
     * @param ctx
     * @throws Exception
     */
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        channelGroup.writeAndFlush("[客户端]" + ctx.channel().remoteAddress() + "离开聊天\n");
        System.out.println("当前在线人数：" + channelGroup.size() + "\n");
    }

    /**
     * 表示Channel处于活动状态，即客户端连接成功
     * 可以提示xx上线
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + "上线了~");
    }

    /**
     * 表示channel处于不活动状态，提示xx离线了
     * @param ctx
     * @throws Exception
     */
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + "离线了~");
    }

    /**
     * 读取数据
     * @param ctx           the {@link ChannelHandlerContext} which this {@link SimpleChannelInboundHandler}
     *                      belongs to
     * @param msg           the message to handle
     * @throws Exception
     */
    @Override
    protected void channelRead0(ChannelHandlerContext ctx, String msg) throws Exception {
        // 获取到当前的channel
        Channel channel = ctx.channel();
        // 先遍历channelGroup ，根据不同的情况，回送不同的信息
        channelGroup.forEach(ch -> {
            if (channel != ch) {// 不是当前的channel，转发消息
                ch.writeAndFlush("[客户端]" + channel.remoteAddress() + "对客户端说：" + msg + "\n");
            } else {
                ch.writeAndFlush("[自己]发送了消息" + msg + "\n");
            }
        });
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        ctx.close();// 关闭通道
    }
}
